package com.wwu.config;

import com.wwu.entity.ScheduledJob;
import lombok.extern.slf4j.Slf4j;
import org.quartz.SchedulerException;
import org.springframework.boot.CommandLineRunner;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @description: 启动项目时预先加载配置文件中的所有定时任务，并根据启动参数启动相应任务
 * CommandLineRunner中的Order表示在项目启动时预先加载的类
 * @author 一蓑烟雨
 * @version 1.0.0
 * @date 2023-04-16 18:13
 */
@Slf4j
@Component
@Order(1)
public class QuartzCommandRunner implements CommandLineRunner {
    @Resource
    private QuartzManager quartzManager;
    @Resource
    private QuartzConfig quartzConfig;

    /**
     * @description: 重写run方法
     * @param [args]
     * @return void
     * @author 一蓑烟雨
     * @date 2023/4/17 15:37
    */
    @Override
    public void run(String... args) throws Exception {
        try {
            //初始化需要启动的任务列表
            List<ScheduledJob> jobList = initSchedule(args);
            for (ScheduledJob scheduledJob : jobList) {
                //执行任务
                quartzManager.startJob(scheduledJob);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * @description: 初始化需要执行的任务列表
     * @param [args]
     * @return java.util.List<com.wwu.entity.ScheduledJob>
     * @author 一蓑烟雨
     * @date 2023/4/17 15:37
    */
    public List<ScheduledJob> initSchedule(String... args) throws SchedulerException {
        List<ScheduledJob> jobs = new ArrayList<ScheduledJob>();
        try {
            log.info("...根据参数加载配置文件quartz.properties中任务列表开始...");
            //1.遍历启动程序时传入的接口类型参数
            String[] typeParam = null;
            for (String arg : args) {
                if (arg.startsWith("type")){
                    typeParam =  arg.substring(5).split(",");
                    break;
                }
            }
            //2.遍历quartz.properties配置文件中任务列表
            for (ScheduledJob scheduledJob : quartzConfig.getScheduledJobs()) {
                //如果启动传入了参数，则按照参数加载启动任务
                if (typeParam != null && typeParam.length > 0) {
                    //遍历参数值和任务列表，若相同则加入到启动任务列表中
                    Arrays.stream(typeParam).forEach(param -> {
                        if (param.trim().equals(scheduledJob.getJobName())) {
                            jobs.add(scheduledJob);
                            log.info("......" + scheduledJob);
                        }
                    });
                } else {
                    //没有启动参数，默认启动配置文件中的所有任务
                    jobs.add(scheduledJob);
                    log.info("......" + scheduledJob);
                }
            }
            log.info("......共加载{}条任务需要执行......", jobs.size());
        } catch (Exception e) {
            e.printStackTrace();
        }
        log.info("...根据参数加载配置文件quartz.properties中任务列表结束...");
        return jobs;
    }
}
